package com.example.wenliangjiudian.interceptor;

import com.example.wenliangjiudian.entity.User;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@Component
public class LoginInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        System.out.println("进入拦截器"+request.getRequestURI());
        //编写业务拦截的规则
        //从session中获取用户的信息
        User admin = (User) request.getSession().getAttribute("user");

        if (admin == null) {
            response.sendRedirect(request.getContextPath() + "/404");
            return false;
        }

        request.setAttribute("nickname", admin.getNickname());
        request.setAttribute("headurl", admin.getHeadurl());
        request.setAttribute("usertype", admin.getUsertype());


        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {
        System.out.println("postHandle");
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        System.out.println("afterCompletion");
    }
}
